#include "ota_http.h"
#include "curl/curl.h"
#include <string.h>
#include "log/log.h"
#include <stdlib.h>

/**
 * @brief curl回调函数 将数据拷贝到传过来的容器中
 *
 * @param contents 响应数据
 * @param size 响应数据大小
 * @param nmemb 响应数据数量
 * @param userp 数据容器
 * @return size_t
 */
static size_t ota_http_write_callback(void *contents, size_t size, size_t nmemb, void *userp)
{
    // 数据长度
    int data_len = size * nmemb;
    // 数据拷贝
    memcpy(userp, contents, data_len);
    // 添加字符串结束符
    ((char *)userp)[data_len] = '\0';
    // 返回数据长度
    return data_len;
}

char *ota_http_getJson(char *url)
{
    // 创建curl对象
    CURL *curl = curl_easy_init();
    // 设置url
    curl_easy_setopt(curl, CURLOPT_URL, url);
    // 设置接收响应数据的回调函数
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, ota_http_write_callback);
    // 向回调函数传递一个数据容器
    char *json_buf = malloc(1024);
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, json_buf);

    // 执行请求
    CURLcode res = curl_easy_perform(curl);
    if (res != CURLE_OK)
    {
        log_error("请求失败: %s", url);
        // 释放curl对象
        curl_easy_cleanup(curl);
        return NULL;
    }

    // 释放curl对象
    curl_easy_cleanup(curl);

    return json_buf;
}

/**
 * @brief
 *
 * @param url   下载地址
 * @param filename  文件路径
 * @return int
 */
int ota_http_download(char *url, char *filename)
{
    // 创建curl对象
    CURL *curl = curl_easy_init();
    // 设置请求的URL
    curl_easy_setopt(curl, CURLOPT_URL, url);
    // 设置接收响应数据的回调函数
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, fwrite);
    // 向回调函数传递一个数据容器
    FILE *file = fopen(filename, "wb");
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, file);

    // 执行请求
    CURLcode res = curl_easy_perform(curl);
    if (res != CURLE_OK)
    {
        log_error("请求失败: %s", url);
        // 释放curl对象
        curl_easy_cleanup(curl);
        // 关闭文件
        fclose(file);
        return -1;
    }

    // 释放curl对象
    curl_easy_cleanup(curl);
    // 关闭文件
    fclose(file);

    return 0;
}
